import service.ai_service
from llm import INPUT_SIZE
from schemas.ai import AIResponse, AIRequest


def init_routes(app):
    @app.post("/scaffolding/api/ai/generateContent", response_model=AIResponse)
    def generateContent(request: AIRequest):
        if len(request.question) > INPUT_SIZE:
            return AIResponse(result="FAILED", answer="Too Many Input Tokens")

        prompt = '''{}
            , 请围绕以上关键字生成一段朋友圈文案，参插一些表情，返回字数在32个汉字以内,表达风格活泼，开朗以及诙谐。
        '''.format(request.question)

        return service.ai_service.generateContent(prompt)
